from .src.nodes import ApplyTKGChromaKeySDXL, ApplyTKGChromaKeyAdvanced

NODE_CLASS_MAPPINGS = {
    "ApplyTKGChromaKeySDXL": ApplyTKGChromaKeySDXL,
    "ApplyTKGChromaKeyAdvanced": ApplyTKGChromaKeyAdvanced,
}
NODE_DISPLAY_NAME_MAPPINGS = {
    "ApplyTKGChromaKeySDXL": "🥚🍚 Apply TKG Noise (SD/SDXL)",
    "ApplyTKGChromaKeyAdvanced": "🥚🍚 Apply TKG Noise Advanced",
}
